



from utils import BatchGenerator,get_file
from plot_batch_img import plot_batch_img,convert_ids_to_class_names_onehot,convert_img_path_to_img_array_list


train_dir = '../MNIST_lit'
train, train_label, val, val_label, class_names_to_ids,ids_to_class_names= get_file(train_dir)

batch_provider = BatchGenerator(train,train_label,32,10)


for i in range(100):
    batch2 = batch_provider.next_batch()
    plot_batch_img(convert_img_path_to_img_array_list(batch2[0]),
                   convert_ids_to_class_names_onehot(batch2[1], ids_to_class_names))





